package com.tom.websocket;

import cn.hutool.core.date.DateUtil;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.util.Date;

/**
 * 处理消息的handler
 * {@link TextWebSocketFrame} 在netty中用于为websocket专门处理文本的对象,frame是消息的载体
 *
 * @author zx
 * @date 2024-10-04 15:55
 */
public class WebSocketCustomerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {
    // 用于记录管理所有客户端的channel;
    private static ChannelGroup clients = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
        // 获取客户端传输过来的消息
        String content = msg.text();
        System.out.println("服务端接收到的消息：" + content);

        // 针对channel 发送消息
        // for (Channel channel : clients) {
        //     TextWebSocketFrame textWebSocketFrame = new TextWebSocketFrame("[服务器接收到的消息:]" + LocalDateTime.now() + "接收到消息,消息为:" + content);
        //     channel.writeAndFlush(textWebSocketFrame);
        // }
        //下面的方式和上面的循环方式一样
        String date = DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss");
        clients.writeAndFlush(new TextWebSocketFrame("[服务器接收到的消息:]" + date+ "接收到消息,消息为:" + content));

    }

    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        // 当客户端连接服务端之后,获取客户端的channel,并且放到ChannelGroup中，并进行管理
        clients.add(ctx.channel());

    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        // 当触发handlerRemoved方法之后，ChannelGroup会自动移除对应客户端的channel
        System.out.println("客户端端开,channel对应的长id:" + ctx.channel().id().asLongText());
        System.out.println("客户端端开,channel对应的端id:" + ctx.channel().id().asShortText());
        //clients.remove(ctx.channel());
    }
}
